package com.base.controller;

import java.util.UUID;
import java.util.concurrent.TimeUnit;

import org.redisson.api.RCountDownLatch;
import org.redisson.api.RLock;
import org.redisson.api.RReadWriteLock;
import org.redisson.api.RSemaphore;
import org.redisson.api.RedissonClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * redis 分布式锁 测试
 * @ClassName: RedissonLockController
 * @Description: 
 * @Author xuyou
 * @DateTime 2020年10月13日 下午4:32:15
 */
@RestController
@RequestMapping("/lock")
public class RedissonLockController {

	@Autowired
	private RedissonClient redisson;
	@Autowired
	private RedisTemplate redisTemplate;
	
	/**
	 * 可重入锁（Reentrant Lock）：最简单的锁
	 * @Title: reentrantLock
	 * @Description: 
	 * @Author xuyou
	 * @DateTime 2020年10月13日 下午4:34:12
	 */
	@GetMapping("/reentrant-lock")
	public void reentrantLock()  {
        // 1. 获取一把锁，只要锁的名字一样，就是同一把锁
        RLock lock = redisson.getLock("my-lock");
        // 2. 加锁
        //lock.lock(); // 堵塞式等待，默认加的锁都是30秒
        //1)、锁的自动续期，如果业务超长，运行期间自动给锁续上新的30秒。不用担心业务时间长，锁自动过期被删掉。
        //2)、加锁的业务只要运行完成，就不会给当前锁续期，即使不手动解锁，锁默认在30秒以后自动删除
        //lock.lock(10, TimeUnit.SECONDS); // 10秒自动解锁，自动解锁时间一定要大于业务的执行时间。
        // 问题：lock.lock(10, TimeUnit.SECONDS); 在锁时间到了以后，不会自动续期。
        // 1. 如果我们传递了锁的超时时间，就发送给redis执行脚本，进行占锁，默认超时就是我们指定的时间
        // 2. 如果我们未指定锁的超时间，就使用30 * 1000【lockWatchdogTimeout 看门狗的默认时间】；
        //      只要占锁成功，就会启动一个定时任务【重新给锁设置过期时间，新的过期时间是看门狗的默认时间】，每个1/3看门狗时间，过期时间就会被重新设置成看门狗默认时间
        //      internalLockLeaseTime【看门狗时间】
        // 最佳实战（生产环境使用的是）
        // lock.lock(30, TimeUnit.SECONDS); 省掉了整个续期操作。因为一个业务超时了肯定就有问题了，没必要接着等待
        try {
            boolean b = lock.tryLock(1, TimeUnit.SECONDS);
            if (!b) {
                System.out.println("加锁失败，直接返回..." + Thread.currentThread().getId());
                return;
            }
            System.out.println("加锁成功，执行业务..." + Thread.currentThread().getId());
            Thread.sleep(10000);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            // 3. 解锁，假设解锁代码没用运行，redisson是不会出现死锁的，因为锁的有效期是30秒，只要业务结束就不会续期。
            System.out.println("释放锁..." + Thread.currentThread().getId());
            //如果lock.lock(10, TimeUnit.SECONDS) 10秒后过期了,之后再解锁的话会报错
            lock.unlock();
        }
    }
	
	
	/**
	 * 读写锁（ReadWriteLock）：保证读取的数据是最新的
	 * @Title: writeValue
	 * @Description: 
	 * @Author xuyou
	 * @DateTime 2020年10月13日 下午5:24:09
	 * @return
	 */
    @GetMapping("/write")
    public String writeValue() {
    	// 业务场景：保证一定能读到最新的是数据，修改期间，写锁是排他锁（互斥锁，独享锁）。读锁是一个共享锁。
    	// 写锁没释放，读锁就必须等待
    	// 写 + 读  等待写锁释放
    	// 写 + 写  堵塞
    	// 读 + 写  有读锁，写也需要等待
    	// 读 + 读  相当于无锁，并发读，只会在redis中记录好，所有当前的读锁。他们会同时加锁成功
    	// 总之：只要有写的存在，无论先写还是后写都会等待。
        RReadWriteLock lock = redisson.getReadWriteLock("rw-lock");
        RLock rLock = lock.writeLock();
        // 修改数据加写锁，读取数据加读锁
        rLock.lock();
        String s = null;
        try {
            System.out.println("写锁加锁成功...." + Thread.currentThread().getId());
            s = UUID.randomUUID().toString();
            Thread.sleep(5000);
            redisTemplate.opsForValue().set("writeValue", s);
        } catch (Exception e) {
             e.printStackTrace();
        } finally {
            rLock.unlock();
            System.out.println("写锁释放...." + Thread.currentThread().getId());
        }
        return s;
    }

    @GetMapping("/read")
    public String readValue() {
        RReadWriteLock lock = redisson.getReadWriteLock("rw-lock");
        RLock rLock = lock.readLock();
        // 修改数据加写锁，读取数据加读锁
        rLock.lock();
        String s = null;
        try {
            System.out.println("读锁加锁成功...." + Thread.currentThread().getId());
            s = (String) redisTemplate.opsForValue().get("writeValue");
            Thread.sleep(1000);
        } catch (Exception e) {
             e.printStackTrace();
        } finally {
            rLock.unlock();
            System.out.println("读锁释放...." + Thread.currentThread().getId());
        }
        return s;
    }

    
    
    
    
    
    
    
    /**
     * 信号量（Semaphore）: 用做分布式限流  也可以用在秒杀 把库存预热
     * @Title: init
     * @Description: 
     * @Author xuyou
     * @DateTime 2020年10月13日 下午5:28:38
     * @return
     */
    @GetMapping("/init")
    public String init() { 
        redisTemplate.opsForValue().set("park", "3");
        return "设置车位";
    }

    /**
	    *   车库停车
	  * 3车位
	     * 信号量也可以用做分布式限流
     */
    @GetMapping("/park")
    public String park() throws InterruptedException {
        RSemaphore park = redisson.getSemaphore("park");
//        park.acquire(); // 获取一个型号，获取一个值，占一个车位，获取不到就堵塞
        boolean b = park.tryAcquire(); // 这个用的比较多，通过这个设置限流
        if (b) {
            // 执行业务
        } else {
            return "error";
        }
        return "ok=>" + b;
    }

    @GetMapping("/go")
    public String go() throws InterruptedException {
        RSemaphore park = redisson.getSemaphore("park");
        park.release(); // 释放一个车位
        return "go";
    }
    
    
    
    
	/**
	 * 闭锁（CountDownLatch）
     * 所有班级的人都走完才能关门
     * @return
     * @throws InterruptedException
     */
    @GetMapping("/lockDoor")
    public String lockDoor() throws InterruptedException {
        RCountDownLatch door = redisson.getCountDownLatch("door");
        door.trySetCount(3);
        door.await(); // 等待闭锁完成，才能走
        return "放假了...";
    }

    @GetMapping("/gogogo/{id}")
    public String gogogo(@PathVariable("id") Long id) {
        RCountDownLatch door = redisson.getCountDownLatch("door");
        door.countDown(); // 计数减一
        return id + "班的人都走了....";
    }
    
    
}
